package com.sh.data.engine.application.model.dto.integration;

import com.sh.data.engine.domain.shims.db.model.TableInfoDomain;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import org.apache.commons.lang3.StringUtils;

import java.util.List;

/**
 * @author: mengzheng.mhc
 * @date: 2024/6/27 16:38
 */
@Data
@Builder
@Schema(description = "获取目标数据源表信息")
@NoArgsConstructor
@AllArgsConstructor
public class GetTargetTablesDTO {

    @Schema(description = "表名列表")
    private List<Table> tables;

    @Data
    @NoArgsConstructor
    @AllArgsConstructor
    public static class Table {
        /**
         * 库名
         */
        private String dbName;

        /**
         * schema
         */
        private String schema;

        /**
         * 表id，hive库有
         */
        private Long tblId;

        /**
         * 表名
         */
        private String tableName;

        /**
         * 表注释
         */
        private String tableComment;

        /**
         * 表类型
         */
        private TableInfoDomain.TblType tblType;

        /**
         * 最新更新时间
         */
        private Long lastDdlTime;

        /**
         * 是否是分区表
         */
        private boolean isPartitionTable;

        private String detailedTblName;

        public enum TblType {
            /**
             * 内表
             */
            MANAGED_TABLE,
            /**
             * 外表
             */
            EXTERNAL_TABLE,
            /**
             * 视图
             */
            VIEW;

            public static TableInfoDomain.TblType from(String tblType) {
                for (TableInfoDomain.TblType value : TableInfoDomain.TblType.values()) {
                    if (StringUtils.equalsIgnoreCase(tblType, value.name())) {
                        return value;
                    }
                }
                return null;
            }
        }
    }
}
